creating a function in python